easy python code for beginners